#!/bin/bash

# 设置环境变量以优化性能
export LD_LIBRARY_PATH=$(python -c "import site; print(site.getsitepackages()[0] + '/nvidia/nvjitlink/lib')"):$LD_LIBRARY_PATH

# 设置 vLLM 优化参数
export VLLM_USE_V1=0  # 使用 V0 引擎以获得更好的兼容性
export CUDA_VISIBLE_DEVICES=0,1,2,3  # 明确指定4张GPU

# 运行优化版本
echo "🚀 启动 4×A100 优化版 MCTS 无代码推理..."
echo "📊 配置：8候选/步，6最大深度，12轮迭代，批处理16"

python main.py \
    --qaf "eval_data/GSM8K_test.json" \
    --custom_cfg "config/sample_mcts_wo_code_optimized.yaml" \
    --output_dir "output" 